package com.cuixichang.minimalism.service;

import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.cuixichang.minimalism.core.mybatis.cnstants.BaseFieldNames;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.util.Assert;
import com.cuixichang.minimalism.mapper.IMenuMapper;
import com.cuixichang.minimalism.transfer.dto.IMenuDTO;
import com.cuixichang.minimalism.transfer.factory.IMenuAssembler;
import com.cuixichang.minimalism.transfer.po.IMenu;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;

import java.time.LocalDateTime;
import java.util.*;
import java.util.function.BiConsumer;
import java.util.stream.Collectors;
import java.util.function.Function;
import org.springframework.stereotype.Service;

/**
 * 系统菜单表表服务实现类
 *
 * @author cuixichang
 * @date 2024-12-18
 */
@Service
public class IMenuServiceImpl implements IMenuService {

    @Autowired
    private IMenuMapper iMenuMapper;

    @Override
    public IMenuDTO selectById(Long id){
        IMenu iMenu = iMenuMapper.selectById(id);
        return IMenuAssembler.INSTANCT.convertDto(iMenu);
    }

    @Override
    public IMenuDTO selectOne(IMenuDTO iMenuDTO){
        Assert.notNull(iMenuDTO,"参数为空,请检查");
        IMenu iMenu = IMenuAssembler.INSTANCT.convertPo(iMenuDTO);
        IMenu result = iMenuMapper.selectOne(new QueryWrapper<>(iMenu));
        return IMenuAssembler.INSTANCT.convertDto(result);
    }

    @Override
    public List<IMenuDTO> selectByIds(List<Long> ids) {
        Assert.notEmpty(ids,"参数为空,请检查");
        List<IMenu> results = iMenuMapper.selectBatchIds(ids);
        return results.stream().map(IMenuAssembler.INSTANCT::convertDto).collect(Collectors.toList());
    }

    @Override
    public List<IMenuDTO> selectList(IMenuDTO iMenuDTO){
        return this.selectList(iMenuDTO, (iMenu, queryWrapper) -> {});
    }

    @Override
    public List<IMenuDTO> selectList(IMenuDTO iMenuDTO, BiConsumer<IMenu, QueryWrapper<IMenu>> consumer){
        Assert.notNull(iMenuDTO,"参数为空,请检查");
        IMenu iMenu = IMenuAssembler.INSTANCT.convertPo(iMenuDTO);
        QueryWrapper<IMenu> queryWrapper = new QueryWrapper<>(iMenu);
        if(Objects.nonNull(consumer)){
            consumer.accept(iMenu, queryWrapper);
        }
        queryWrapper.orderBy(true,true,BaseFieldNames.SORT).orderBy(true,false,BaseFieldNames.UPDATE_TIME);
        List<IMenu> results = iMenuMapper.selectList(queryWrapper);
        return results.stream().map(IMenuAssembler.INSTANCT::convertDto).collect(Collectors.toList());
    }

    @Override
    public Page<IMenuDTO> selectPage(IMenuDTO iMenuDTO,long pageSize,long pageNo){
        return this.selectPage(iMenuDTO, pageSize, pageNo, (iMenu, queryWrapper) -> {});
    }

    @Override
    public Page<IMenuDTO> selectPage(IMenuDTO iMenuDTO,long pageSize,long pageNo, BiConsumer<IMenu, QueryWrapper<IMenu>> consumer){
        Assert.notNull(iMenuDTO,"参数为空,请检查");
        IMenu iMenu = IMenuAssembler.INSTANCT.convertPo(iMenuDTO);
        QueryWrapper<IMenu> queryWrapper = new QueryWrapper<>(iMenu);
        if(Objects.nonNull(consumer)) consumer.accept(iMenu, queryWrapper);
        queryWrapper.orderBy(true,true,BaseFieldNames.SORT).orderBy(true,false,BaseFieldNames.UPDATE_TIME);
        Page<IMenu> page = iMenuMapper.selectPage(new Page<>(pageNo,pageSize),queryWrapper);
        Page<IMenuDTO> result = Page.of(page.getCurrent(),page.getSize(),page.getTotal());
        result.setRecords(page.getRecords().stream()
                .map(IMenuAssembler.INSTANCT::convertDto).collect(Collectors.toList()));
        return result;
    }

    @Override
    public int deleteById(Long id){
        return iMenuMapper.deleteById(id);
    }

    @Override
    public int deleteByIds(List<Long> ids){
        return iMenuMapper.deleteBatchIds(ids);
    }

    @Override
    public int deleteByCondition(IMenuDTO iMenuDTO){
        return deleteByCondition(iMenuDTO, (iMenu, queryWrapper)-> {});
    }

    @Override
    public int deleteByCondition(IMenuDTO iMenuDTO, BiConsumer<IMenu, QueryWrapper<IMenu>> consumer){
        Assert.notNull(iMenuDTO,"参数为空,请检查");
        IMenu iMenu = IMenuAssembler.INSTANCT.convertPo(iMenuDTO);
        QueryWrapper<IMenu> queryWrapper = new QueryWrapper<>(iMenu);
        if(Objects.nonNull(consumer)) {
            consumer.accept(iMenu, queryWrapper);
        }
        return iMenuMapper.delete(queryWrapper);
    }

    @Override
    public IMenuDTO insert(IMenuDTO iMenuDTO){
        return insertHasOption(iMenuDTO, (IMenuDTO) -> null);
    }

    @Override
    public IMenuDTO insertHasOption(IMenuDTO iMenuDTO, Function<IMenuDTO,IMenuDTO> preOption){
        Assert.notNull(iMenuDTO,"参数为空,请检查");
        iMenuDTO.check();
        IMenu iMenu = IMenuAssembler.INSTANCT.convertPo(iMenuDTO);
        Long createLoginId = iMenu.getCreateLoginId();
        Assert.notNull(createLoginId,"创建人ID为空,请检查");
        iMenu.setUpdateLoginId(createLoginId);
        LocalDateTime localDateTime = LocalDateTime.now();
        iMenu.setCreateTime(localDateTime);
        iMenu.setUpdateTime(localDateTime);
        Boolean isEnable = iMenu.getIsEnable();
        if(Objects.isNull(isEnable)) iMenu.setIsEnable(true);
        Integer sort = iMenu.getSort();
        if(Objects.isNull(sort)) iMenu.setSort(0);
        if(Objects.nonNull(preOption)){
            IMenuDTO apply = preOption.apply(iMenuDTO);
            if(Objects.nonNull(apply)) {
              return apply;
            }
        }
        int insert = iMenuMapper.insert(iMenu);
        Assert.isTrue(insert>0,"插入失败");
        return IMenuAssembler.INSTANCT.convertDto(iMenu);
    }

    @Override
    public int updateById(IMenuDTO iMenuDTO){
         return updateById(iMenuDTO, (iMenu, updateWrapper) -> {});
    }

    @Override
    public int updateById(IMenuDTO iMenuDTO, BiConsumer<IMenu, UpdateWrapper<IMenu>> consumer){
        Assert.notNull(iMenuDTO,"参数为空,请检查");
        Long id = iMenuDTO.getId();
        Assert.notNull(id,"主键为空,请检查");
        IMenu iMenu = IMenuAssembler.INSTANCT.convertPo(iMenuDTO);
        Long createLoginId = iMenu.getCreateLoginId();
        Assert.notNull(createLoginId,"创建人ID为空,请检查");
        iMenu.setCreateLoginId(null);
        iMenu.setUpdateLoginId(createLoginId);
        iMenu.setUpdateTime(LocalDateTime.now());
        UpdateWrapper<IMenu> updateWrapper = new UpdateWrapper<IMenu>().eq(BaseFieldNames.ID, id);
        if(Objects.nonNull(consumer)) consumer.accept(iMenu,updateWrapper);
        return iMenuMapper.update(iMenu,updateWrapper);
    }
}
